import pandas as pd

number = 509
score = pd.read_csv('output/predictions/compare_keepsearched05_bs_top10.csv').loc[number:number]
label = pd.read_csv('datasets/body_title/test-uncased.csv').loc[number:number]['title']
res1 = pd.read_csv('output/predictions/t5-small-title-generation-lowercase-alldata-bs8-402-23/predictions-bs.csv').loc[
       number:number].values.tolist()
res2 = pd.read_csv('output/predictions/t5-small-title-generation-lowercase-alldata-bs8-402-23/predictions-bs-top10-keepsearched05.csv').loc[
       number:number].values.tolist()

print("score:")
print(score)
print("target result:")
print(label)
print(100 * '-')

print('{:<80}   {:<80}'.format("beam search", "similarity"))
for i in range(len(res1)):
    for j in range(11):
        print('{:<80}   {:<80}'.format(res1[i][j], res2[i][j]))

print(100 * '-')

for i in range(len(res1)):
    for j in range(101):
        print('{:<80}'.format(res1[i][j]))

# for i in range(len(res2)):
#     for j in range(11):
#         print('{:<60}'.format(res2[i][j]))
